#pragma once

#include <iostream>
#include <unistd.h>
#include <sys/types.h>
#include <sys/wait.h>
#include <sys/stat.h>
#include <fcntl.h>
#include "../comm/util.hpp"
#include "../comm/log.hpp"

using namespace ns_util;
using namespace ns_log;

// 负责客户端提交的代码的编译
namespace ns_compiler
{
    class Compiler
    {
    public:
        Compiler()
        {
        }
        ~Compiler()
        {
        }
        // 编译,成功返回true,失败返回false
        // file_name：文件名，不包含路径和后缀
        // eg. file_name:123
        static bool Compile(const std::string &file_name)
        {
            // 对客户端提交的代码编译通过程序替换exec实现
            pid_t pid = fork();
            if (pid < 0)
            {
                LOG(ERROR) << "内部错误，创建子进程失败" << std::endl;
                return false;
            }
            else if (pid == 0)
            {
                umask(0);
                int _compile_error = open(PathUtil::CompileError(file_name).c_str(), O_CREAT | O_WRONLY, 0644);
                if (_compile_error < 0)
                {
                    LOG(WARNING) << "没有成功地形成.compile_error文件" << std::endl;
                    exit(1);
                }
                // 重定向标准错误到_stderr
                dup2(_compile_error, 2);
                // 子进程: 调用编译器，完成对代码的编译工作
                // g++ -o target src -std=c++11
                execlp("g++", "g++", "-o", PathUtil::Exe(file_name).c_str(), PathUtil::Src(file_name).c_str(), 
                "-std=c++11","-D","COMPILER_ONLINE", nullptr);
                LOG(ERROR) << "启动g++编译器失败，可能是参数错误" << std::endl;
                exit(2);
            }
            else
            {
                waitpid(pid, nullptr, 0);
                // 编译是否成功,就看有没有形成对应的可执行程序
                if (FileUtil::IsFileExists(PathUtil::Exe(file_name)))
                {
                    LOG(INFO) << PathUtil::Src(file_name) << "编译成功!" << std::endl;
                    return true;
                }
            }
            // 走到这里说明编译失败
            LOG(ERROR) << "编译失败，没有形成可执行程序" << std::endl;
            return false;
        }
    };

};